from rest_framework.permissions import BasePermission

# 必须先走完（通过）认证才能走到鉴权这儿
# 定义了两个权限类，一个SVIP（user_type=3）,一个是VIP(user_type=2)

class SVIPPermission(BasePermission):
    """SVIP鉴权类"""
    message = "必须是svip才能访问"  # 自定义错误信息
    def has_permission(self,request,view):
        """这个request.user就是自定义认证类返回的那个元组中的第一个元
        必须认证类返回了那个元组，我们这儿才能使用上request.user当前登录用户
        """
        if request.user.user_type != 3:
            return False
        # 就返回是否有权限就行了不用抛异常.
        return True


class VIPPermission(BasePermission):
    """VIP鉴权类"""
    def has_permission(self,request,view):
        if request.user.user_type == 2:
            return False
        return True